package qc.module.ehs.mapper;

import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import org.mapstruct.IterableMapping;
import org.mapstruct.Mapper;
import org.mapstruct.Mapping;
import org.mapstruct.Named;
import org.mapstruct.factory.Mappers;
import qc.common.core.enums.EhsUnitTypeEnum;
import qc.module.ehs.dto.unitcategory.TplUnitCategoryDto;
import qc.module.ehs.dto.unitcategory.TplUnitCategoryTreeTableItemDto;
import qc.module.ehs.entity.EhsTplUnitcategory;

import java.util.List;

/**
 * EhsTplUnitcategoryMapper
 *
 * @author QuCheng Tech
 * @create 2024/5/22
 */
@Mapper
public interface EhsTplUnitcategoryMapper {
    EhsTplUnitcategoryMapper MAPPER = Mappers.getMapper(EhsTplUnitcategoryMapper.class);

    @Mapping(target = "uctype", source = "uctype.index")
    TplUnitCategoryDto toDto(EhsTplUnitcategory en);

    @Mapping(target = "uctype", ignore = true)
    EhsTplUnitcategory toEntity(TplUnitCategoryDto dto);

    List<TplUnitCategoryDto> toDtoList(List<EhsTplUnitcategory> ens);

    @Mapping(target = "uctype", source = "uctype.index")
    @Mapping(target = "children", ignore = true)
    @Named("toTreeTableItemDto")
    TplUnitCategoryTreeTableItemDto toTreeTableItemDto(EhsTplUnitcategory en);

    @IterableMapping(qualifiedByName = "toTreeTableItemDto")
    List<TplUnitCategoryTreeTableItemDto> toTreeTableItemDtoList(List<EhsTplUnitcategory> list);

    /**
     * 新增修改DTO转换为Entity，特殊处理
     *
     * @param dto 新增DTO
     * @return Entity
     * @author QuCheng Tech
     * @since 2024/5/22
     */
    default EhsTplUnitcategory dtoToEntity(TplUnitCategoryDto dto) {
        EhsTplUnitcategory en = EhsTplUnitcategoryMapper.MAPPER.toEntity(dto);

        //处理组织类型，非空时进行转换，空值时使用默认
        Integer uctype = null;
        if (StringUtils.isNotBlank(dto.getUctype())) {
            try {
                uctype = Integer.parseInt(dto.getUctype());
            } catch (NumberFormatException exception) {
                //转换错误不处理，使用默认状态
            }
        }
        en.setUctype(getEhsUnitTypeEnum(uctype));

        return en;
    }

    /**
     * 根据枚举key获取枚举对象
     *
     * @param index
     * @return
     */
    static EhsUnitTypeEnum getEhsUnitTypeEnum(Integer index) {
        if (index != null && index > 0x0) {
            for (EhsUnitTypeEnum item : EhsUnitTypeEnum.values()) {
                if (item.getIndex() == index) {//判断序号使用getIndex，判断内容使用getName
                    return item;
                }
            }
        }

        //单元类型默认为1-区域
        return EhsUnitTypeEnum.AREA;
    }
}
